package cn.tedu.v4;

import java.io.*;
import java.net.BindException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

/**
 * Version 4 多人聊天室 ---消息转发
 * 建议：单线聊天，多人聊天，多人聊天+QQ模式 【消息转发功能】
 * 拓展功能：建议增加第三个版本 Version3
 *
 *消息转发步骤：服务器给所有的客户端写出内容，客户端之间相互收到内容
 * 1.集合存储所有写出的客户端--服务器端写出给所有客户端
 * 2.广播消息给所有的客户端---broadcastMessage 广播留言
 *  注意：写在ClientHandler上，必须同步方法
 * 3.将客户端的输出流添加 allOut集合中
 *   注意：卸载ClientHandler的构造器--所有客户端具备功能：消息转发
 * 4.广播消息给所有的客户端
 *   注意：写在ClientHandler的run
 * 5.客户端退出，从 allOut 集合中移除
 *   注意：写在ClientHandler的 run--finally【一定溢出链接】
 */
public class Server {
    private ServerSocket serverSocket;
    private static final Integer PORT=6666;
    //1.私有属性：集合存储所有写出的客户端--服务器端写出给所有客户端
    private List<PrintWriter> allOut=new ArrayList<>();

    public Server() {
        try {
            serverSocket = new ServerSocket(PORT);
        } catch (BindException e) {
            System.out.println("端口号重复出申请，建议换个端口号");
        } catch (IOException e) {
            System.out.println("申请端口号失败....");
        }
    }

    public void start() {
        try {
            while (true) {

                System.out.println("----等待客户端链接-----");
                Socket socket = serverSocket.accept();

                new Thread(new ClientHandler(socket)).start();
            }
        } catch (IOException e) {
            System.out.println("服务器端通讯断开了");
        } finally {
            if (serverSocket != null && !serverSocket.isClosed()) {
                try {
                    serverSocket.close();
                } catch (IOException e) {
                    System.out.println("服务器关闭失败");
                }
            }

        }

    }

    public static void main(String[] args) throws IOException {new Server().start();}
   //2.广播消息给所有的客户端---功能 broadcastMessage 广播留言
    //2.1 broadcastMessage功能：提供转发给客户端消息【加同步锁】
    //    理解：来一个客户端，只允许当前客户端转发消息
    //2.2 String message 参数：转发消息---代表：消息
    private synchronized void broadcastMessage(String message){
        //2.3增强for：【理解：来很多客户端，很多的客户端遍历出来，消息转发出去】
        for (PrintWriter pw: allOut){//allOut保存所有链接的客户端
            pw.println("消息广播："+message);//理解：消息转发出去
        }
    }
    private class ClientHandler implements Runnable {
        private Socket socket;
        private String ip;
        private BufferedReader br;
        private PrintWriter pw;

        public ClientHandler(Socket s) {
            this.socket = s;
            ip = socket.getInetAddress().getHostAddress();

            try {
                 br = new BufferedReader(new InputStreamReader(socket.getInputStream(), "UTF8"));
                 pw = new PrintWriter(new BufferedWriter(new OutputStreamWriter(socket.getOutputStream(), "UTF8")), true);
                 //3.所有客户端具备消息转发
                //理解：客户端写出文字，服务器负责转发文字
                //监视同步对象：allOut---存所有客户端--
                //客户端存在线程中---执行存在执行顺序不一致情况
                synchronized (allOut) {
                    allOut.add(pw);//客户端写出文字，存集合中
                }

            } catch (IOException e) {
                System.out.println("客户端断开链接~~");
            }
        }

        public void run() {

            try {
                new Thread(() -> {
                    String message;
                    try {
                        while ((message = br.readLine()) != null) {
                            System.out.println("客户端说：" + message);
                            System.out.println("服务器端请输入：");
                            //4.广播消息给所有的客户端
                            //broadcastMessage转发；写出文本内容
                            //message 客户端发给都武器的内容
                            //把当前客户端写出内容，转发给所有的客户端
                            broadcastMessage(message);
                        }
                    } catch (IOException e) {
                        System.out.println("客户端读入失败");
                    }
                }).start();
                Scanner input = new Scanner(System.in);
                while (true) {
                    System.out.println("服务器请输入");
                    String say = input.nextLine();
                    pw.println("服务器说：" + say);
                }
            } catch (NullPointerException e) {
                System.out.println("IO 流传入对象，" +
                        "有对象为null!立马核对，" +
                        "建议：查看try块中对象可能赋值为null");
            }  finally {
                //5.一旦有客户端断开，客户端从allOut 集合中移除
                //必须同步块，结束客户端不一致
                synchronized (allOut) {//监视allOut集合中客户端有没有断开
                    allOut.remove(pw);//allOut集合客户端断开链接，移除写出流
                }

                if (br!= null)
                    try {
                        br.close();
                    } catch (IOException e) {
                        System.out.println("结束通讯：关闭失败！");
                    }
                //输出流关闭
                if (pw!=null) pw.close();
                if (socket != null && !socket.isClosed()) {
                    try {
                        socket.close();
                    } catch (IOException e) {
                        System.out.println("通讯关闭失败");
                    }
                }
            }

        }

    }
}